from ninja import Schema, Field


class AddVipPackageSchema(Schema):
    name: str = Field(default="", description="名称")
    day: int = Field(default=0, description="天数")
    order: int = Field(default=1, description="排序")
    price: float = Field(default=0.0, description="价格")
    brief_introduction: str = Field(default="", description="简介")
    equity_statement: str = Field(default="", description="权益说明")


class AddAgentPackageSchema(Schema):
    name: str = Field(default="", description="名称")
    water_card_number: int = Field(default=0, description="水卡数量")
    price: float = Field(default=0.0, description="价格")
    channel: list[int] = Field(default=[], description="渠道")
    username: str = Field(default=None, description="联系人")
    phone_number: str = Field(default=None, description="联系方式")
    address: str = Field(default=None, description="联系地址")
